from artifacts.base.base_artifacts import Artifacts
from artifacts.base.attr_type import AttrType


class Flower(Artifacts):
    def get_name(self):
        return '花'

    def get_main_attr_weight_map(self):
        return {
            AttrType.HP: 100
        }


class Plume(Artifacts):
    def get_name(self):
        return '毛'

    def get_main_attr_weight_map(self):
        return {
            AttrType.ATK: 100
        }


class Clock(Artifacts):
    def get_name(self):
        return '表'

    def get_main_attr_weight_map(self):
        return {
            AttrType.HP_P: 1334,
            AttrType.ATK_P: 1333,
            AttrType.DEF_P: 1333,
            AttrType.ENERGY: 500,
            AttrType.ELEMENTAL: 500,
        }  # sum = 5000


class Cup(Artifacts):
    def get_name(self):
        return '杯'

    def get_main_attr_weight_map(self):
        return {
            AttrType.HP_P: 850,
            AttrType.ATK_P: 850,
            AttrType.DEF_P: 800,
            AttrType.ELEMENTAL: 100,
            AttrType.PYRO_DMG: 200,
            AttrType.ELECTRO_DMG: 200,
            AttrType.CRYO_DMG: 200,
            AttrType.HYDRO_DMG: 200,
            AttrType.ANEMO_DMG: 200,
            AttrType.GEO_DMG: 200,
        }  # sum = 3800


class Head(Artifacts):
    def get_name(self):
        return '头'

    def get_main_attr_weight_map(self):
        return {
            AttrType.HP_P: 1100,
            AttrType.ATK_P: 1100,
            AttrType.DEF_P: 1100,
            AttrType.ELEMENTAL: 200,
            AttrType.CRIT_DMG: 500,
            AttrType.CRIT_RATE: 500,
            AttrType.HEALING: 500,
        }  # sum = 5000
